# 被测试的函数
def add(a, b):
    return a + b

# 导入被测试的函数
# from your_module import add

# 定义测试函数
def test_add_positive_numbers():
    assert add(1, 2) == 3

def test_add_negative_numbers():
    assert add(-1, -2) == -3

def test_add_mixed_numbers():
    assert add(1, -2) == -1